find square python program